package com.example.jingleai.repository;

import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Component
public class InMemoryChatHistoryRepository implements ChatHistoryRepository {
    // 一个简单的内存中的聊天历史记录存储库，用于存储和检索聊天记录。
    // 该存储库使用一个 Map 来存储聊天记录，其中键是聊天类型（如“chat”或“conversation”），值是聊天记录列表。
    private final Map<String, List<String>> chatHistory = new HashMap<>();

    @Override
    public void save(String type, String chatId) {
        // 如果键不存在，则创建一个空的聊天记录列表
        List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new ArrayList<>());
        if (chatIds.contains(chatId)) {
            return;
        }
        chatIds.add(chatId);
    }

    @Override
    public List<String> getChatIds(String type) {
        // 返回与给定类型关联的聊天记录列表，如果不存在，则返回一个空的列表。
        return chatHistory.getOrDefault(type, List.of());
    }
}
